package test.mp.domain.dto;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import test.mp.domain.po.User;
import test.mp.domain.vo.UserVO;

import java.util.Collections;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * @ClassName PageDTO
 * @Description 分页返回值
 * @Author wxx
 * @Date 2024/08/25/22:08
 * @Version 1.0
 */
@Data
@ApiModel(description = "分页结果")
public class PageDTO<T> {
    @ApiModelProperty("总条数")
    private Long total;
    @ApiModelProperty("总页数")
    private Long pages;
    @ApiModelProperty("集合")
    private List<T> list;


    /**
     *
     * @param p
     * @param clazz
     * @return
     * @param <PO>
     * @param <VO>
     */
    public static<PO,VO> PageDTO<VO> of(Page<PO> p, Class<VO> clazz){
        // 3.封装VO结果
        PageDTO<VO> pageDTO = new PageDTO<>();
        // 总条数
        pageDTO.setTotal(p.getTotal());
        // 总页数
        pageDTO.setPages(p.getPages());
        // 3.3 当前页数据
        List<PO> records = p.getRecords();
        if(CollUtil.isEmpty(records)){
            pageDTO.setList(Collections.emptyList());
            return pageDTO;
        }
        // 3.4 拷贝user的VO
        pageDTO.setList(BeanUtil.copyToList(records, clazz));
        return pageDTO;
    }


    /**
     * 自定义转换
     * @param p
     * @param convertor
     * @return
     * @param <PO>
     * @param <VO>
     */
    public static<PO,VO> PageDTO<VO> of(Page<PO> p, Function<PO,VO> convertor){
        // 3.封装VO结果
        PageDTO<VO> pageDTO = new PageDTO<>();
        // 总条数
        pageDTO.setTotal(p.getTotal());
        // 总页数
        pageDTO.setPages(p.getPages());
        // 3.3 当前页数据
        List<PO> records = p.getRecords();
        if(CollUtil.isEmpty(records)){
            pageDTO.setList(Collections.emptyList());
            return pageDTO;
        }
        // 3.4 拷贝user的VO
        pageDTO.setList(records.stream().map(convertor).collect(Collectors.toList()));
        return pageDTO;
    }

}
